
from lenet import Lenet
from tensorflow.keras.utils import plot_model


def main():
    model = Lenet.build(width=28, height=28, depth=1, classes=10)
    plot_model(model, to_file="model.jpg", show_shapes=True)


if __name__ == '__main__':
    main()
